#!/bin/bash

# 设置conda环境
echo "正在设置conda环境..."
source $(conda info --base)/etc/profile.d/conda.sh
conda activate fastvlm || { echo "激活环境失败，尝试重新创建环境"; conda create -n fastvlm python=3.10 -y && conda activate fastvlm; }

# 安装项目依赖
echo "正在安装项目依赖..."
pip install -e .

# 下载预训练模型（如果需要）
if [ ! -d "checkpoints/llava-fastvithd_0.5b_stage3" ]; then
  echo "正在下载预训练模型..."
  bash get_models.sh
else
  echo "模型已存在，跳过下载步骤"
fi

echo "\n设置完成！\n"
echo "要运行推理，请使用以下命令："
echo "python predict.py --model-path ./checkpoints/llava-fastvithd_0.5b_stage3 --image-file <图片路径> --prompt \"描述这张图片。\""